# coding=utf-8
"""
该脚本用于替换VOC格式的XMl中的路径
"""
import os
import os.path
import xml.dom.minidom
# 替换xml文件存放路径
file_path = r"C:\Users\HsuLee\Desktop\VOCdevkit\VOC2007\Annotations"
# 替换图片保存路径
picture_path = r"C:\Users\HsuLee\Desktop\VOCdevkit\VOC2007\JPEGImages"
files = os.listdir(file_path)  # 得到文件夹下所有文件名称
s = []
for xmlFile in files:  # 遍历文件夹
    if not os.path.isdir(xmlFile):  # 判断是否是文件夹,不是文件夹才打开
        print(xmlFile)
        # xml文件读取操作
        # 将获取的xml文件名送入到dom解析
        dom = xml.dom.minidom.parse(os.path.join(file_path, xmlFile))  ###最核心的部分,路径拼接,输入的是具体路径
        root = dom.documentElement
        # 获取标签对path之间的值
        original_path = root.getElementsByTagName('path')
        # 原始信息
        p0=original_path[0]
        path0=p0.firstChild.data   #原始路径
        print(path0)
        # 修改
        jpg_name=path0.split('\\')[-1]   #获取图片名
        modify_path=picture_path+'\\'+jpg_name   #修改后path
        p0.firstChild.data=modify_path

        # 保存修改到xml文件中
        with open(os.path.join(file_path, xmlFile), 'w') as fh:
            dom.writexml(fh)
            print('修改path OK!')



